export dataset_name=/home/user4/workspace/proteingym-benchmark/data
# Tranception_Small Tranception_Medium Tranception_Large
# example for dataset evaluation
CUDA_VISIBLE_DEVICES=3 python src/Tranception/predict.py \
--dataset_input $dataset_name \
--model_dir src/Tranception/models \
--model_locate Tranception_Large \
--batch_size_inference 64 \
--scoring_window sliding

# example for single protein evaluation
CUDA_VISIBLE_DEVICES=0 python src/Tranception/predict.py \
--model_dir src/Tranception/models \
--model_locate Tranception_Large \
--batch_size_inference 32 \
--scoring_window sliding \
--fasta_file None \
--DMS_file None \
--MSA_file None \
--MSA_weight_file None \
--MSA_start 1 \
--MSA_end 559 \
--inference_time_retrieval \
--retrieval_inference_weight 0.6